package dfs;

/**
 * @author wcj
 * @Date 2021/7/6 20:26
 * @Version 1.0
 */
public class FriendCircle {

    private static int count;
    private static int[][] tmp;

    /**
     * 朋友圈：friends[i][j]表示第i个人和第j个人的关系，1表示是朋友，0表示不是，朋友关系具有传递性，求朋友圈个数
     * 深搜：
     * ①递归方法的设计：dfs(int i,int j)其中i表示第i个人
     * ②递归出口：越界
     * ③递归组合方式：上下左右四个方向深搜
     *
     * @param friends
     * @return
     */
    public static int friendCircle(int[][] friends) {
        count = 0;
        tmp = friends;
        for (int i = 0; i < tmp.length; i++) {
            for (int j = 0; j < tmp.length; j++) {
                if (tmp[i][j] == 1) {
                    dfs(i, j);
                    count++;
                }
            }
        }
        return count;
    }

    public static void dfs(int i, int j) {
        if (i < 0 || i >= tmp.length || j < 0 || j >= tmp.length) {
            return;
        }
        if (tmp[i][j] == 1) {
            tmp[i][j] = 2;
            dfs(i + 1, j);
            dfs(i - 1, j);
            dfs(i, j - 1);
            dfs(i, j + 1);
        }
    }

}
